#ifndef _COMM_HPP_
#define _COMM_HPP_

#include<iostream>
#include<cassert>
#include<cstring>
#include<cerrno>
#include<sys/ipc.h>
#include<sys/shm.h>
#include<cstdlib>
#include<unistd.h>

#define PATHNAME "."//ftok的第一个参数,是一个合法路径
#define PROJ_IO 0X666//ftok的第二个参数
#define MAX_SIZE 4096//共享内存的大小

key_t getkey()
{
    key_t k = ftok(PATHNAME, PROJ_IO);
    if(k < 0)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(1);
    }
    return k;
}
int getshmhelper(key_t k, int flags)
{
    int shmid = shmget(k, MAX_SIZE, flags);
    if(shmid < 0)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(2);
    }
    return shmid;
}

int getshm(key_t k)//获取共享内存
{
    return getshmhelper(k, IPC_CREAT);//没有就创建，有就获取
}
int createshm(key_t k)//创建共享内存
{
    return getshmhelper(k, IPC_CREAT | IPC_EXCL | 0600);//没有创建，有就报错,这里创建内存需要给对应的权限
}

void delshm(int shmid)
{
    if(shmctl(shmid, IPC_RMID, nullptr) == -1)
    {
        std::cerr << errno << ":" << strerror(shmid) << std::endl;
        exit(3);
    }
}
void* attachshm(int shmid)
{
    void* p = shmat(shmid, nullptr, 0);
    if((long long)p == -1L)//因为linux系统是64位，一个地址是8个字节，所以要变成8个字节大小的数据类型做对比
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
        exit(4);
    }
    return p;
}
void detachshm(void* p)
{
    if(shmdt(p) == -1)
    {
        std::cerr << errno << ":" << strerror(errno) << std::endl;
    }
}

#endif
